#ifndef CONSTANTS_H
#define CONSTANTS_H

#include <map>
#include <vector>
#include <string>

// 数据文件名
const std::vector<std::string> GESTURES = {"zuoxie", "youxie", "hen", "shu", "(", ")", "idle"};
// 数据文件类别，onehot编码，1值位置表示类别
const std::map<std::string, std::vector<int>> GESTURES_TO_Y = {
    {"zuoxie", {1,0,0,0,0,0,0}},
    {"youxie", {0,1,0,0,0,0,0}},
    {"hen", {0,0,1,0,0,0,0}},
    {"shu", {0,0,0,1,0,0,0}},
    {"(", {0,0,0,0,1,0,0}},
    {")", {0,0,0,0,0,1,0}},
    {"idle", {0,0,0,0,0,0,1}}
};
// 一个sample数据形状
const int T = 10;
const int V = 6;
const int TYPE_NUM = 7;
const int AUG_STEPS = 5;
// echonet 参数
const int N_INTERNAL_UNITS = 50;
const float SPECTRAL_RADIUS = 0.59;
const float LEAK = 0.6;
const float CONNECTIVITY = 0.25;
const float INPUT_SCALING = 0.1;
const float NOISE_LEVEL = 0.0;
const bool CIRCLE = false;
// 是否加载现有模型
const bool IF_LOAD_MODEL = true;

#endif
